Chip 2007 January, February, March & April/Chip-Cover-CD-2007-02.iso/Pakiet bezpieczenstwa/mini Pentoo LiveCD 2006.1/mpentoo-2006.1.iso/livecd.squashfs/usr/lib/python2.4/test/test_uu.pyo

Python Compiled Bytecode | 2005-10-18 | 6KB | 190 lines

# Source Generated with Decompyle++
# File: in.pyo (Python 2.4)

'''
Tests for uu module.
Nick Mathewson
'''
import unittest
from test import test_support
import sys
import os
import uu
import cStringIO
import uu
from StringIO import StringIO
plaintext = 'The smooth-scaled python crept over the sleeping dog\n'
encodedtext = 'M5&AE(\'-M;V]T:"US8V%L960@<\'ET:&]N(&-R97!T(&]V97(@=&AE(\'-L965P\n(:6YG(&1O9PH '
encodedtextwrapped = 'begin %03o %s\n' + encodedtext.replace('%', '%%') + '\n \nend\n'

class UUTest(unittest.TestCase):

    def test_encode(self):
        inp = cStringIO.StringIO(plaintext)
        out = cStringIO.StringIO()
        uu.encode(inp, out, 't1')
        self.assertEqual(out.getvalue(), encodedtextwrapped % (438, 't1'))
        inp = cStringIO.StringIO(plaintext)
        out = cStringIO.StringIO()
        uu.encode(inp, out, 't1', 420)
        self.assertEqual(out.getvalue(), encodedtextwrapped % (420, 't1'))


    def test_decode(self):
        inp = cStringIO.StringIO(encodedtextwrapped % (438, 't1'))
        out = cStringIO.StringIO()
        uu.decode(inp, out)
        self.assertEqual(out.getvalue(), plaintext)
        inp = cStringIO.StringIO('UUencoded files may contain many lines,\n' + "even some that have 'begin' in them.\n" + encodedtextwrapped % (438, 't1'))
        out = cStringIO.StringIO()
        uu.decode(inp, out)
        self.assertEqual(out.getvalue(), plaintext)


    def test_truncatedinput(self):
        inp = cStringIO.StringIO('begin 644 t1\n' + encodedtext)
        out = cStringIO.StringIO()

        try:
            uu.decode(inp, out)
            self.fail('No exception thrown')
        except uu.Error:
            e = None
            self.assertEqual(str(e), 'Truncated input file')



    def test_missingbegin(self):
        inp = cStringIO.StringIO('')
        out = cStringIO.StringIO()

        try:
            uu.decode(inp, out)
            self.fail('No exception thrown')
        except uu.Error:
            e = None
            self.assertEqual(str(e), 'No valid begin line found in input file')




class UUStdIOTest(unittest.TestCase):

    def setUp(self):
        self.stdin = sys.stdin
        self.stdout = sys.stdout


    def tearDown(self):
        sys.stdin = self.stdin
        sys.stdout = self.stdout


    def test_encode(self):
        sys.stdin = cStringIO.StringIO(plaintext)
        sys.stdout = cStringIO.StringIO()
        uu.encode('-', '-', 't1', 438)
        self.assertEqual(sys.stdout.getvalue(), encodedtextwrapped % (438, 't1'))


    def test_decode(self):
        sys.stdin = cStringIO.StringIO(encodedtextwrapped % (438, 't1'))
        sys.stdout = cStringIO.StringIO()
        uu.decode('-', '-')
        self.assertEqual(sys.stdout.getvalue(), plaintext)



class UUFileTest(unittest.TestCase):

    def _kill(self, f):

        try:
            f.close()
        except (SystemExit, KeyboardInterrupt):
            raise
        except:
            pass


        try:
            os.unlink(f.name)
        except (SystemExit, KeyboardInterrupt):
            raise
        except:
            pass



    def setUp(self):
        self.tmpin = test_support.TESTFN + 'i'
        self.tmpout = test_support.TESTFN + 'o'


    def tearDown(self):
        del self.tmpin
        del self.tmpout


    def test_encode(self):

        try:
            fin = open(self.tmpin, 'wb')
            fin.write(plaintext)
            fin.close()
            fin = open(self.tmpin, 'rb')
            fout = open(self.tmpout, 'w')
            uu.encode(fin, fout, self.tmpin, mode = 420)
            fin.close()
            fout.close()
            fout = open(self.tmpout, 'r')
            s = fout.read()
            fout.close()
            self.assertEqual(s, encodedtextwrapped % (420, self.tmpin))
        finally:
            self._kill(fin)
            self._kill(fout)



    def test_decode(self):

        try:
            f = open(self.tmpin, 'wb')
            f.write(encodedtextwrapped % (420, self.tmpout))
            f.close()
            f = open(self.tmpin, 'rb')
            uu.decode(f)
            f.close()
            f = open(self.tmpout, 'r')
            s = f.read()
            f.close()
            self.assertEqual(s, plaintext)
        finally:
            self._kill(f)



    def test_decodetwice(self):

        try:
            f = cStringIO.StringIO(encodedtextwrapped % (420, self.tmpout))
            f = open(self.tmpin, 'rb')
            uu.decode(f)
            f.close()
            f = open(self.tmpin, 'rb')
            self.assertRaises(uu.Error, uu.decode, f)
            f.close()
        finally:
            self._kill(f)




def test_main():
    test_support.run_unittest(UUTest, UUStdIOTest, UUFileTest)

if __name__ == '__main__':
    test_main()